@inproceedings{jang2017categorical,
  title={Categorical reparameterization with gumbel-softmax},
  author={Jang, Eric and Gu, Shixiang and Poole, Ben},
  booktitle={International Conference on Learning Representations},
  year={2017},
  url={https://openreview.net/pdf?id=rkE3y85ee}
}